package com.engine.cn.leetcode;

import java.util.Stack;

/**
 * 设计一个支持 push ，pop ，top 操作，并能在常数时间内检索到最小元素的栈。
 * 实现 MinStack 类:
 * MinStack() 初始化堆栈对象。
 * void push(int val) 将元素val推入堆栈。
 * void pop() 删除堆栈顶部的元素。
 * int top() 获取堆栈顶部的元素。
 * int getMin() 获取堆栈中的最小元素。
 * 解决思路
 * 方法一：辅助栈法（推荐）
 * 使用两个栈：主栈存储所有元素，辅助栈存储当前最小值
 * 当压入新元素时：
 * 主栈正常压入
 * 如果辅助栈为空或新元素 ≤ 辅助栈顶元素，则压入辅助栈
 * 弹出元素时：
 * 主栈正常弹出
 * 如果弹出的元素等于辅助栈顶元素，则辅助栈也弹出
 * getMin 直接返回辅助栈顶元素
 */
public class MinStack {
    private Stack<Integer> mainStack;
    private Stack<Integer> slaveryStack;

    public MinStack() {
        mainStack = new Stack<>();
        slaveryStack = new Stack<>();
    }

    public void push(int val) {
        mainStack.push(val);
        if (slaveryStack.isEmpty()) {
            if (val < slaveryStack.peek()) {
                slaveryStack.push(val);
            }
        }
    }

    public void pop() {
        if (mainStack.pop().equals(slaveryStack.peek())) {
            slaveryStack.pop();
        }
    }

    public int top() {
        return mainStack.peek();
    }

    public int getMin() {
        return slaveryStack.peek();
    }
}
